package com.lebang.medicalInquiry.model.chat;

import lombok.AllArgsConstructor;
import lombok.Data;

@Data
@AllArgsConstructor
public class Message {

    private String role;

    private String content;

    public static Message ofSystem(String content) {
        return new Message(Role.SYSTEM.getValue(), content);
    }

    public  static Message ofUser(String content) {
        return new Message(Role.USER.getValue(), content);
    }

    public enum Role {
        SYSTEM("system"),
        USER("user"),
        ASSISTANT("assistant");

        private String value;

        public String getValue() {
            return this.value;
        }

        private Role(String value) {
            this.value = value;
        }
    }
}
